from llama_index.llms.azure_openai import AzureOpenAI

llm = AzureOpenAI(
    engine="gpt-4o",
    api_key="c8575027653b42b1b47747f0b4ab135b",
    api_version="2023-05-15",
    azure_endpoint="https://menshen.test.xdf.cn"
)

# Send a completion call to generate an answer
print('Sending a test completion job')
start_phrase = '''
你是一个优秀的测试经理，请编写一份测试用例，该用例的产品需求如下：
1.业务背景
    该业务属于CRM系统中的体验课业务，需求是用户在体验课业务中进行多种操作，操作完毕后
    需要把这些操作的结果，以消息的方式发送给下游数据采集方，下游数据采集方是：大数据项目组；
2.用户在体验课业务中进行多种操作的具体行为有四种：
（1）第一种是报名（也叫邀约）；
（2）第二种是到访（也叫签到）；
（3）第三种是确访（也叫确认到访）；
（4）第四种是更换场次（相当于是取消场次再重新邀约）。
3.业务要求是：把上面的这四种操作，在完毕后以消息的方式发送给大数据项目组。
4.用例格式要求
（1）需要用MarkDown格式输出测试用例；
（2）表格的标题分别是：序号，用例标题，用例描述，测试步骤，执行结果；总共五列；'''

response = llm.complete(start_phrase)
print(response)
print(start_phrase + response.raw.choices[0].message.content)